#pragma once

#include <iostream>
#include <string>
#include <signal.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

const static std::string defaultpath = "/";
const static std::string defaultdev = "/dev/null";

void Daemon(bool ischdir, bool isclose)
{
    // 1.忽略可能引起异常的信号
    signal(SIGCHLD, SIG_IGN);
    signal(SIGPIPE, SIG_IGN);

    // 2.创建子进程，保证不是组长进程
    if (fork() > 0)
        exit(0);

    // 3.创建新会话
    setsid();

    // 4.改变工作目录
    if (ischdir)
        chdir(defaultpath.c_str());

    // 5.守护进程不需要用户的输入、输出、错，关闭相关fd或者重定向
    if (isclose)
    {
        ::close(0);
        ::close(1);
        ::close(2);
    }
    else
    {
        int fd = open(defaultdev.c_str(), O_RDWR);
        // 重定向
        dup2(fd, 0);
        dup2(fd, 1);
        dup2(fd, 2);
        ::close(fd);
    }
}